#include "mystdio.h"
#include <assert.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>

#define FILE_MODE 0666

// 打开文件【"w"、"r"、"a"】
_FILE *_fopen(const char *filename, const char *flag)
{
    assert(filename);
    assert(flag);

    int f = 0;
    int fd = -1; // 文件描述符
    if (strcmp(flag, "w") == 0)
    {
        f = O_CREAT | O_WRONLY | O_TRUNC; // O_TRUNC：清空和写入
        fd = open(filename, f, FILE_MODE);
    }
    else if (strcmp(flag, "r") == 0)
    {
        f = O_RDONLY;
        fd = open(filename, f);
    }
    else if (strcmp(flag, "a") == 0)
    {
        f = O_CREAT | O_WRONLY | O_APPEND; // O_TRUNC：清空和写入
        fd = open(filename, f, FILE_MODE);
    }
    else
    {
        return NULL;
    }

    if (fd == -1)
        return NULL; // 打开文件失败

    _FILE *fp = (_FILE *)malloc(sizeof(_FILE));
    if (fp == NULL)
        return NULL;

    // 填充字段
    fp->fileno = fd;
    fp->flag = FLUSH_LINE;
    fp->in_pos = 0;
    return fp;
}

// 写入文件
int _fwrite(_FILE *fp, const char *s, int len)
{
    // “abcd\n”
    memcpy(&fp->in_buffer[fp->in_pos], s, len);
    fp->in_pos += len;

    if (fp->flag & FLUSH_NOW)
    {
        write(fp->fileno, fp->in_buffer, fp->in_pos);
        fp->in_pos = 0;
    }
    else if (fp->flag & FLUSH_LINE)
    {
        if (fp->in_buffer[fp->in_pos - 1] == '\n')
        {
            write(fp->fileno, fp->in_buffer, fp->in_pos);
            fp->in_pos = 0;
        }
    }
    else if (fp->flag & FLUSH_ALL)
    {
        if (fp->in_pos == SIZE)
        {
            write(fp->fileno, fp->in_buffer, fp->in_pos);
            fp->in_pos = 0;
        }
    }

    return len;
}

void _fflush(_FILE *fp)
{
    if (fp->in_pos > 0)
    {
        write(fp->fileno, fp->in_buffer, fp->in_pos);
        fp->in_pos = 0;
    }
}

// 关闭文件
void _fclose(_FILE *fp)
{
    if (fp == NULL)
        return;
    _fflush(fp);
    close(fp->fileno);
    free(fp);
}